
from . import common


def get_api_result_大中型水库名录(str_token, date_from, date_to):
    # 请求参数 大中型水库名录
    body_dict = {
        "startTime": date_from,
        "endTime": date_to,
        "pageSize": 10,
        "pageNumber": 1,
    }
    api_result = common.exec_water_api(str_token, "https://sldn.slt.zj.gov.cn:30003/service/v2/api/provider/query/18f47585", None, body_dict)
    print("大中型水库名录 请求ok")
    return api_result


def get_api_result_大中型水库基础信息(str_token, date_from, date_to):
    # 请求参数 大中型水库名录
    body_dict = {
        "startTime": date_from,
        "endTime": date_to,
        "pageSize": 10,
        "pageNumber": 1,
    }
    api_result = common.exec_water_api(str_token, "https://sldn.slt.zj.gov.cn:30003/service/v2/api/provider/query/7b9dde4d", None, body_dict)
    print("大中型水库名录 请求ok")
    return api_result


def get_api_result_河道水位监测数据(date_from, date_to):
    body_dict = {
        "startTime": date_from,
        "endTime": date_to,
        "stcds": "13813",
        "sttp": "ZZ",
        # "group": 1,
        # "pageSize": 2,
        # "pageNumber": 1,
    }
    str_token, dest_url = common.get_generate_token("f896e295")
    api_result = common.exec_water2_api(str_token, dest_url, None, body_dict, "message")
    print("河道水位监测数据 请求结束")
    return api_result


def get_api_result_水文站点数据(date_from, date_to):
    # 水文站点数据 请求参数
    body_dict = {
        "startTime": date_from,
        "endTime": date_to,
        "sttp": "ZZ",
        "pageSize": "10",
        "pageNumber": "1",
    }
    str_token, station_url = common.get_generate_token("55e51c47")
    print("token is " + str_token)
    api_result = common.exec_water2_api(str_token, station_url, None, body_dict, "message.rows")
    print("水文站点数据 请求结束")
    return api_result
